package com.generator;


import cn.hutool.setting.dialect.Props;
import com.mybatisflex.codegen.Generator;
import com.mybatisflex.codegen.config.GlobalConfig;
import com.mybatisflex.codegen.dialect.JdbcTypeMapping;
import com.zaxxer.hikari.HikariDataSource;

import java.time.LocalDateTime;
import java.util.Date;
/**
 * 代码生成器 完全自动
 * */
public class GeneratorNew {

    public static void main(String[] args) {
        String xmName="/";//项目名称
        String projectPath = System.getProperty("user.dir")+"/"+xmName;//项目打开的时候 根路径
        //获取配置文件的内容
//        Props props = new Props(projectPath + "/src/main/resources/application.properties");
//        String active = props.getStr("spring.profiles.active");
        Props application = new Props(projectPath + "/src/main/resources/application.properties");
        String dataSourceName = application.getStr("spring.datasource.username");
        String dataSourcePassword = application.getStr("spring.datasource.password");
        String dataSourceUrl = application.getStr("spring.datasource.url");
        //获取该配置文件自定义信息
//        String mk = application.getStr("shop.mk");//模块名称
        String tab = application.getStr("shop.tables");//表名
        String auth = application.getStr("shop.author");//作者名称
        String basePackage = application.getStr("shop.basepackage");//第一个位置根路径

        //配置数据源
        HikariDataSource dataSource = new HikariDataSource();
        dataSource.setJdbcUrl(dataSourceUrl);//+"&useInformationSchema=true"
        dataSource.setUsername(dataSourceName);
        dataSource.setPassword(dataSourcePassword);

        //创建配置内容，两种风格都可以。
        GlobalConfig globalConfig = createGlobalConfigUseStyle1(projectPath,tab,auth,basePackage);

        //通过 datasource 和 globalConfig 创建代码生成器
        Generator generator = new Generator(dataSource, globalConfig);
        //生成代码
        generator.generate();
    }

    public static GlobalConfig createGlobalConfigUseStyle1(String projectPath,String tab,String auth,String basePackage) {

        //需要生成的表
        String[] tables = tab.split(",");
        //创建配置内容
        GlobalConfig globalConfig = new GlobalConfig();

        //设置根目录
        globalConfig.setSourceDir(projectPath+"/src/main/java");

        //设置根包
        globalConfig.setBasePackage(basePackage);

        //设置只生成哪些表
        globalConfig.setGenerateTable(tables);

        //设置实体类的包路径 在自动生成的service中找实体类路径的时候，import com.zq.mybatis.sys.entity;否则不设置会异常
        //globalConfig.setEntityPackage("com.zq.mybatis."+mk+".entity");
        //globalConfig.setMapperPackage("com.zq.mybatis."+mk+".mapper");//import 是根据这个路径找到该mapper

        //设置生成 entity 并启用 Lombok
        globalConfig.setEntityGenerateEnable(true);
        globalConfig.setEntityWithLombok(true);
        globalConfig.setEntityOverwriteEnable(true);

        //设置生成 mapper
        globalConfig.setMapperGenerateEnable(true);

        //设置xml
        globalConfig.setMapperXmlPath(projectPath.concat("/src/main/resources/mapper/"));
        globalConfig.setMapperXmlGenerateEnable(true);

        //设置def
        globalConfig.setTableDefGenerateEnable(true);

        //设置service
        globalConfig.setServiceGenerateEnable(true);

        // impl
        globalConfig.setServiceImplGenerateEnable(true);

        //设置controller
        globalConfig.setControllerGenerateEnable(true);


        //可以单独配置某个列
        JdbcTypeMapping.registerMapping(LocalDateTime.class, Date.class);
        globalConfig.getJavadocConfig().setAuthor(auth);//作者名字 可以随便起名

        return globalConfig;
    }
}

